package com.example.demo.dao;

import com.example.demo.entity.pojo.AgentMessage;
import org.apache.ibatis.annotations.Param;
import java.util.List;

/**
 * Agent对话消息表(AgentMessage)表数据库访问层
 *
 * @author makejava
 * @since 2025-07-13 18:30:30
 */
public interface AgentMessageDao {

    /**
     * 通过ID查询单条数据
     *
     * @param id 主键
     * @return 实例对象
     */
    AgentMessage queryById(Integer id);

    /**
     * 根据会话ID查询消息列表
     *
     * @param conversationId 会话ID
     * @param limit 限制数量
     * @return 消息列表
     */
    List<AgentMessage> queryByConversationId(@Param("conversationId") Integer conversationId, 
                                           @Param("limit") Integer limit);

    /**
     * 根据会话ID查询最近的消息
     *
     * @param conversationId 会话ID
     * @param limit 限制数量
     * @return 消息列表
     */
    List<AgentMessage> queryRecentMessages(@Param("conversationId") Integer conversationId, 
                                         @Param("limit") Integer limit);

    /**
     * 根据消息类型查询
     *
     * @param conversationId 会话ID
     * @param messageType 消息类型
     * @param limit 限制数量
     * @return 消息列表
     */
    List<AgentMessage> queryByMessageType(@Param("conversationId") Integer conversationId,
                                        @Param("messageType") String messageType,
                                        @Param("limit") Integer limit);

    /**
     * 查询包含工具调用的消息
     *
     * @param conversationId 会话ID
     * @return 消息列表
     */
    List<AgentMessage> queryMessagesWithTools(@Param("conversationId") Integer conversationId);

    /**
     * 新增数据
     *
     * @param agentMessage 实例对象
     * @return 影响行数
     */
    int insert(AgentMessage agentMessage);

    /**
     * 批量新增消息
     *
     * @param messages 消息列表
     * @return 影响行数
     */
    int insertBatch(@Param("messages") List<AgentMessage> messages);

    /**
     * 修改数据
     *
     * @param agentMessage 实例对象
     * @return 影响行数
     */
    int update(AgentMessage agentMessage);

    /**
     * 通过主键删除数据
     *
     * @param id 主键
     * @return 影响行数
     */
    int deleteById(Integer id);

    /**
     * 删除会话的所有消息
     *
     * @param conversationId 会话ID
     * @return 影响行数
     */
    int deleteByConversationId(@Param("conversationId") Integer conversationId);

    /**
     * 统计会话消息数量
     *
     * @param conversationId 会话ID
     * @return 消息数量
     */
    long countByConversationId(@Param("conversationId") Integer conversationId);

    /**
     * 统计会话token使用量
     *
     * @param conversationId 会话ID
     * @return token总数
     */
    long sumTokensByConversationId(@Param("conversationId") Integer conversationId);
}
